Python interview question and answer